package cn.iocoder.yudao.module.pay.api.wallet;

import cn.hutool.core.lang.Assert;
import cn.iocoder.yudao.framework.common.enums.pay.PayWalletBizTypeEnum;
import cn.iocoder.yudao.module.member.api.user.MemberUserApi;
import cn.iocoder.yudao.module.pay.api.wallet.dto.PayWalletAddBalanceReqDTO;
import cn.iocoder.yudao.module.pay.dal.dataobject.wallet.PayWalletDO;
import cn.iocoder.yudao.module.pay.service.wallet.PayWalletService;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;

/**
 * 钱包 API 实现类
 *
 * @author 芋道源码
 */
@Service
public class PayWalletApiImpl implements PayWalletApi {

    @Resource
    private PayWalletService payWalletService;
    @Resource
    private MemberUserApi memberUserApi;

    @Override
    public void addWalletBalance(PayWalletAddBalanceReqDTO reqDTO) {
        if (reqDTO.getUserType() == null) {
            reqDTO.setUserType(memberUserApi.getUserType(reqDTO.getUserId()));
        }
        // 创建或获取钱包
        PayWalletDO wallet = payWalletService.getOrCreateWallet(reqDTO.getUserId(), reqDTO.getUserType(), reqDTO.getWalletType());
        Assert.notNull(wallet, "钱包({}/{}/{})不存在", reqDTO.getUserId(), reqDTO.getUserType(),  reqDTO.getWalletType());

        // 增加余额
        PayWalletBizTypeEnum bizType = PayWalletBizTypeEnum.getEnum(reqDTO.getBizType());
        payWalletService.addWalletBalance(wallet.getId(), reqDTO.getBizId(), bizType, reqDTO.getPrice(),"");
    }

}
